@echo off

REM 使用Python 3.10创建虚拟环境并安装Qwen3依赖

echo ====================================
echo Qwen3-0.6B 环境配置脚本
echo 使用Python 3.10创建虚拟环境

echo 正在检查Python 3.10是否安装...
py -3.10 --version >nul 2>&1
if %errorlevel% neq 0 (
    echo 未找到Python 3.10！请先安装Python 3.10
    echo 下载地址：https://www.python.org/downloads/release/python-31011/
    echo.
    echo 如遇Python版本问题，请确保使用Python 3.10版本
    pause
    exit /b 1
)

echo 已找到Python 3.10，版本信息：
py -3.10 --version

echo.
echo 正在使用Python 3.10创建虚拟环境...
py -3.10 -m venv qwen3_env

if %errorlevel% neq 0 (
    echo 创建虚拟环境失败！
    echo 请检查是否有足够的权限或磁盘空间。
    pause
    exit /b 1
)

echo 虚拟环境创建成功！
echo.
echo 激活虚拟环境...
call qwen3_env\Scripts\activate

echo 升级pip...
python -m pip install --upgrade pip

echo.
echo 安装PyTorch（CPU版本）...
python -m pip install torch torchvision torchaudio

echo.
echo 安装Transformers和其他依赖...
python -m pip install transformers>=4.51.0 accelerate

echo.
echo ====================================
echo 环境配置完成！
echo ====================================
echo.
echo 您现在可以运行：python index.py
echo 输入'quit'可以退出交互式对话
echo.
echo 注意：首次运行会自动下载模型文件，可能需要一些时间。
echo 请确保您的网络连接正常。
echo.
pause